public class Test3 {
    //矩阵区域和
    //链接: https://leetcode.cn/problems/matrix-block-sum/description/

    public int[][] matrixBlockSum(int[][] mat, int k) {
        //用二维前缀和
        int m = mat.length;
        int n = mat[0].length;
        int[][] dp = new int[m+1][n+1];   //方便初始化   +1
        //初始化dp
        for(int i = 1; i <= m ;i++) {
            for(int j = 1;j <= n; j++) {
                dp[i][j] = dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1] + mat[i-1][j-1];  //这里要减1  因为扩大了数组大小
            }
        }

        //使用一下dp数组
        int[][] answer = new int[m][n];
        for(int i = 0; i < m;i++) {
            for(int j = 0; j < n; j++) {
                int x1 = Math.max(0,i-k) + 1;  //这里使用max是防止下标越界  且  要加1   这里从0下标开始的   但是 你要找的dp数组是从1开始的   所以要加1
                int y1 = Math.max(0,j-k) +1;
                int x2 = Math.min(m-1,i+k) +1;
                int y2 = Math.min(n-1,j+k) +1;
                answer[i][j] = dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1];
            }
        }
        return answer;

    }
    public static void main(String[] args) {

    }
}
